package com.jojo.intermediate.day3_array;

import java.util.ArrayList;
import java.util.List;

/**
 * 119、杨辉三角2
 *
 * 给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。
 * 在「杨辉三角」中，每个数是它左上方和右上方的数的和。
 *
 * 示例 1:
 * 输入: rowIndex = 3
 * 输出: [1,3,3,1]
 *
 * 示例 2:
 * 输入: rowIndex = 0
 * 输出: [1]
 *
 * 示例 3:
 * 输入: rowIndex = 1
 * 输出: [1,1]
 *
 * 进阶：
 * 你可以优化你的算法到 O(rowIndex) 空间复杂度吗？
 */
public class YHTriangleSecond {

    /** myCode 将杨辉三角存入list中 */
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> lists = new ArrayList<>();
        for (int i = 0;i <= rowIndex;i++){
            List<Integer> list = new ArrayList<>();
            for (int j = 0;j <= i;j++){
                if (j == 0 || j == i){
                    list.add(1);
                }else {
                    list.add(lists.get(i - 1).get(j - 1) + lists.get(i - 1).get(j));
                }
            }
            lists.add(list);
        }
        return lists.get(rowIndex);
    }

    /** pe. 递推 O(rowIndex)空间复杂度，从右往左 */
    public List<Integer> getRow2(int rowIndex) {
        List<Integer> row = new ArrayList<>();
        row.add(1);
        for (int i = 1; i <= rowIndex; ++i) {
            row.add(0);
            for (int j = i; j > 0; --j) {
                row.set(j, row.get(j) + row.get(j - 1));
            }
        }
        return row;
    }

    /** pe. 线性递推 */
    public List<Integer> getRow3(int rowIndex) {
        List<Integer> row = new ArrayList<>();
        row.add(1);
        for (int i = 1; i <= rowIndex; ++i) {
            row.add((int) ((long) row.get(i - 1) * (rowIndex - i + 1) / i));
        }
        return row;
    }

    public static void main(String[] args) {
        YHTriangleSecond second = new YHTriangleSecond();
        List<Integer> row = second.getRow2(4);
        System.out.println(row);
    }
}
